import java.util.*;

/**
 * 面试题 10.02. 变位词组
 * https://leetcode-cn.com/problems/group-anagrams-lcci/
 */
public class Solutions_mianshi_10_02 {
    public static void main(String[] args) {
        String[] strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
        // output: {{"ate","eat","tea"}, {"nat","tan"}, {"bat"}}

        List<List<String>> result = groupAnagrams(strs);
        System.out.println(result);
    }

    /**
     * 解法二：哈希表（优化版）
     * 解题思路：所有的变位词按字典序排序后，得到的字符串相同
     */
    public static List<List<String>> groupAnagrams(String[] strs) {
        if (strs == null || strs.length < 1) {
            return new ArrayList<>();
        }
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] arr = str.toCharArray();
            // 排序
            Arrays.sort(arr);
            String s = String.valueOf(arr);
            if (!map.containsKey(s)) {
                // 将变位词都记录到 map 中，键为排序后的字符串
                map.put(s, new ArrayList<>());
            }
            map.get(s).add(str);
        }
        return new ArrayList<>(map.values());
    }

    /**
     * 解法一：哈希表
     * 解题思路：所有的变位词按字典序排序后，得到的字符串相同
     */
    public static List<List<String>> groupAnagrams2(String[] strs) {
        List<List<String>> res = new ArrayList<>();
        if (strs == null || strs.length < 1) {
            return res;
        }
        int[] counts = new int[26];
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] arr = str.toCharArray();
            for (char c : arr) {
                counts[c - 'a'] ++;
            }
            int idx = 0;
            // 计数排序
            for (int i = 0; i < 26; i++) {
                while (counts[i] > 0) {
                    arr[idx++] = (char)(i + 'a');
                    counts[i] --;
                }
            }
            String s = String.valueOf(arr);
            if (!map.containsKey(s)) {
                // 将变位词都记录到 map 中，键为排序后的字符串
                map.put(s, new ArrayList<>());
            }
            map.get(s).add(str);
        }
        for (String str : map.keySet()) {
            res.add(map.get(str));
        }
        return res;
    }
}
